class Prompt:
    def translation_prompt(self, text, language):
        TRANSLATIONPROMPT = (
            f"你是一个翻译家，擅长将文本翻译成 {language}。请确保翻译保持流畅和自然，同时保留原文的含义。\n"
            "以下是一段json字符串:\n"
            f"{text}\n"
            f"请你将其中的text字段的内容翻译为{language}\n"
            "要求：\n"
            f"1.尽量避免直译，而是在理解原文内容的基础上进行意译，使翻译更加贴近{language}的表达习惯。\n"
            "2.只用返回这段json字符串的翻译结果，不需要返回其他内容。如果是text字段的内容是数字符号单位，请直接返回原json。\n"
            "3.注意保留<sub></sub>标签"
        )
        return TRANSLATIONPROMPT

if __name__ == '__main__':
    import json
    from src.tools.chain import gen_chain
    prompt = Prompt()
    text = {"index": 0, "text": "C级、C级背景下的局部A级"}
    json_text = json.dumps(text, ensure_ascii=False)
    query = prompt.translation_prompt(json_text, "英语")
    res = gen_chain(query)
    print(res)
